#include<stdio.h>
#include<string.h>                                                              
#include<unistd.h>
#include<pthread.h>
#include<fcntl.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<stdlib.h>
#include<sys/epoll.h>
#include<errno.h>

#define PORT 8000
#define BUFSIZE 4
#define EVENTSIZE 1024

int create_socket(int port)
{
    int lfd = socket(AF_INET,SOCK_STREAM,0);
    if(lfd == -1)
    {
        perror("socket error");
        exit(1);
    }
    struct sockaddr_in serveraddr;
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(port);
    serveraddr.sin_addr.s_addr = 0; // ip 任意
    // 绑定IP和端口号
    bind(lfd,(struct sockaddr*)&serveraddr,sizeof(serveraddr));
    // 主动套接字转为被动套接字
    listen(lfd,64);
    return lfd; 
}

int main(int argc, char* argv[])
{
    struct sockaddr_in cliaddr;
    int lfd = create_socket(PORT);
    char buf[BUFSIZE];
    int wait_ret;
    int cfd;
    int read_ret;

    // 创建epoll对象
    int epfd = epoll_create(64);
    if(epfd == -1)
    {
        perror("epoll_create error");
        exit(1);
    }

    // 用来存储触发事件的文件描述符
    struct epoll_event events[EVENTSIZE]; 
    // 创建一个结构体，将lfd 文件描述符信息存入
    struct epoll_event event; 
    event.events = EPOLLIN; // 监听读事件
    event.data.fd = lfd; // 存lfd文件描述符
    // 将lfd添加到红黑树中进行监听读事件
    epoll_ctl(epfd,EPOLL_CTL_ADD,lfd,&event); 

    while(1)
    {
        // 参数2是一个传出参数，将触发事件的文件描述符从链表中读取出来
        wait_ret = epoll_wait(epfd,events,EVENTSIZE,-1);
        if(wait_ret == -1)
        {
            perror("epoll_wait error");
            exit(1);
        }
        for(int i=0;i<wait_ret;i++)
        {
            if(events[i].events & EPOLLIN) // 判断是否触发读事件
            {
                if(events[i].data.fd == lfd) // 触发事件的文件描述符是lfd
                {
                    int caddrlen = sizeof(cliaddr);
                    cfd = accept(lfd,(struct sockaddr*)&cliaddr,&caddrlen);
                    if(cfd == -1)
                    {
                        perror("accept error");
                        exit(1);
                    }
                    // 取出客户端端口号
                    int port = ntohs(cliaddr.sin_port);
                    // 取出客户端ip
                    char dst[64];
                    inet_ntop(AF_INET,&cliaddr.sin_addr.s_addr,dst,sizeof(dst));
                    printf("客户端%s 建立连接,端口号为：%d\n",dst,port);

                    // 将文件描述符设置为非阻塞
                    int flags = fcntl(cfd, F_GETFL);
                    flags |= O_NONBLOCK;
                    fcntl(cfd, F_SETFL, flags);

                    event.events = EPOLLIN | EPOLLET; // 边沿触发，默认水平触发
                    event.data.fd = cfd;
                    epoll_ctl(epfd,EPOLL_CTL_ADD,cfd,&event); // 将cfd添加到红黑树中进行监听读事件
                }
                else
                {
                    while(1)
                    {
                        read_ret = read(events[i].data.fd,buf,BUFSIZE);
                        if(read_ret == -1)
                        {
                            if(errno == EAGAIN)
                            {
                                break;
                            }
                            perror("read error");
                            exit(1);
                        }
                        else if(read_ret == 0)
                        {
                            printf("客户端退出\n");
                            epoll_ctl(epfd,EPOLL_CTL_DEL,events[i].data.fd,NULL); // 将文件描述符从红黑树中删除 ，最后一个参数传空就行，删除不需要管结构体中的数据是什么
                            close(events[i].data.fd); // 关闭文件描述符
                            break;
                        }
                        else
                        {
                            write(events[i].data.fd,buf,read_ret);
                            write(STDOUT_FILENO,buf,read_ret);
                        }
                    }
                }
            }
        }
    }
    return 0;
}

